# -*- coding:utf-8 -*-
# @Time:2024/4/1921:01
# @Author:miuzg
# @FileName:new test2.py
# @Software:PyCharm

import torch
import torch.nn as nn  # 优化器
from torch.utils.data import DataLoader  # 数据加载
from torchvision import datasets,transforms  # 数据集和数据变换
from tqdm import tqdm
import os
import sys
from model.cnn import simplecnn
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
train_tansformer = transforms.Compose([
    transforms.Resize(224,224),  # 将数据裁剪为224*224
    transforms.ToTemsor,
    transforms.Normalize((0.5,0.5,0.5),(0.5,0.5,0.5))  # 标准化
])